import puppeteer from 'puppeteer-core';
import path from 'path';

async function scrapeDJI() {
  try {
    // 设置输出目录
    const outputDir = './output';
    
    // 创建输出目录
    const fs = await import('fs');
    if (!fs.existsSync(outputDir)) {
      fs.mkdirSync(outputDir, { recursive: true });
    }

    // 启动浏览器
    const browser = await puppeteer.launch({
      executablePath: '/usr/bin/chromium',
      args: [
        '--no-sandbox',
        '--disable-setuid-sandbox',
        '--disable-dev-shm-usage'
      ]
    });

    // 创建新页面
    const page = await browser.newPage();
    
    // 设置视窗大小
    await page.setViewport({ width: 1920, height: 1080 });

    // 访问大疆网站
    console.log('正在访问大疆官网...');
    await page.goto('https://www.dji.com/cn', {
      waitUntil: 'networkidle0',
      timeout: 60000
    });

    // 保存页面为 HTML
    const html = await page.content();
    console.log('正在保存页面内容...');
    
    // 将内容写入文件
    const htmlPath = path.join(outputDir, 'dji.html');
    await fs.promises.writeFile(htmlPath, html);
    console.log(`页面已保存到 ${htmlPath}`);

    // 截图
    const screenshotPath = path.join(outputDir, 'dji.png');
    await page.screenshot({ path: screenshotPath, fullPage: true });
    console.log(`截图已保存到 ${screenshotPath}`);

    // 关闭浏览器
    await browser.close();
    console.log('任务完成！');

  } catch (error) {
    console.error('发生错误：', error);
  }
}

scrapeDJI(); 